import os
import allure
import pyscreenshot as ImageGrab
import time

currTime = time.strftime("%m%d%H%M%S")
def getScreenShot(funcName):
    # 定义截图保存的路径
    filename = os.path.abspath(os.path.join(os.getcwd(), ".")) + r'/Screen_shot/' + funcName+' ' + currTime + '.png'
    # 这里我猜应该是".."取上层目录
    im = ImageGrab.grab()  # 创建grab对象
    im.save(filename)  # 保存当前页面的截图到 filename
    im.close()  # 销毁 grab 对象
    # 把截图添加到 allure 附件中（测试报告中会带此截图）
    allure.attach.file(source=filename, name=f'{funcName}+{currTime}', attachment_type=allure.attachment_type.PNG)